package com.example.mq_sm.mqserver.core;

import com.example.mq_sm.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 这个类表示一个存储消息的队列
 */
@Data
public class MSGQueue {
    // 表示队列的身份标识
    private String name;
    // 标识队列是否持久化，true 表示持久化保存，false 表示不持久化
    private boolean durable = false;
    // 这个属性为 true，表示这个队列只能被一个消费者使用（别人用不了）。如果为 false 则大家都能使用
    // 这个 独占 功能，后续的代码没有实现这个功能（RabbitMQ 中有）
    private boolean exclusive = false;
    // 为true没人使用就会自动删除，false则不会自动删除
    // 这个属性暂时先列在这里，后续的代码没有实现这个功能（RabbitMQ 中有）
    private boolean autoDelete = false;
    // 拓展参数，后续的代码没有实现这个功能（RabbitMQ 中有）
    private Map<String, Object> arguments = new HashMap<>();
    // 当前队列都有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 记录当前取到了第几个消费者，方便实现轮询策略
    private AtomicInteger consumerSeq = new AtomicInteger(0);
    // 添加一个新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv){
        consumerEnvList.add(consumerEnv);
    }
    // 订阅者的删除暂时先不考虑

    // 挑选一个订阅者，用来处理当前的消息（按照轮询的方式）
    public ConsumerEnv chooseConsumer(){
        if (consumerEnvList.size() == 0){
            // 该队列没有人订阅
            return null;
        }
        // 计算一下当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }

    // 需要传递到数据库而重新写的 getter 和 setter 方法
    public String getArguments(){
        // 吧当前的 arguments 参数，从 Map 转换成 String （JSON）
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        // 如果代码真的发生异常了，就返回一个空的 JSON 字符串
        return "{}";
    }

    public void setArguments(String argumentsJson){
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            // 第二个参数是用来描述当前 json 字符串要转换成的对象是什么类型的
            // 一般情况下，只需 String.class 这种即可
            // 但是如果是一些较为复杂的类型，则需要 TypeReference 来进行操作
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<Map<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    public void setArguments(Map<String, Object> arguments){
        this.arguments = arguments;
    }

    public Object getArguments(String key){
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }
}
